/**   
* @Title: ImageUtils.java 
* @Package com.git.gather 
* @Description: 缩略图实现，将图片(jpg、bmp、png、gif等等)真实的变成想要的大小
* @author fanqinghui100@126.com   
* @date 2014年7月24日 上午10:50:28 
* @version V1.0   
*/ 
package com.git.gather;

import java.awt.Image;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;

import javax.imageio.ImageIO;

import com.sun.image.codec.jpeg.JPEGCodec;
import com.sun.image.codec.jpeg.JPEGImageEncoder;

/**
 * @author fanqinghui100@126.com
*/
public class ImageUtils{

	private File file = null; // 文件对象
	private String inputDir; // 输入图路径
	private String outputDir; // 输出图路径
	private String inputFileName; // 输入图文件名
	private String outputFileName; // 输出图文件名
	private int outputWidth = 320; // 默认输出图片宽
	private int outputHeight = 480; // 默认输出图片高
	private boolean proportion = true; // 是否等比缩放标记(默认为等比缩放)

	public ImageUtils() { // 初始化变量
		inputDir = "";
		outputDir = "";
		inputFileName = "";
		outputFileName = "";
		outputWidth = 100;
		outputHeight = 100;
	}

	public void setInputDir(String inputDir) {
		this.inputDir = inputDir;
	}

	public void setOutputDir(String outputDir) {
		this.outputDir = outputDir;
	}

	public void setInputFileName(String inputFileName) {
		this.inputFileName = inputFileName;
	}

	public void setOutputFileName(String outputFileName) {
		this.outputFileName = outputFileName;
	}

	public void setOutputWidth(int outputWidth) {
		this.outputWidth = outputWidth;
	}

	public void setOutputHeight(int outputHeight) {
		this.outputHeight = outputHeight;
	}

	public void setWidthAndHeight(int width, int height) {
		this.outputWidth = width;
		this.outputHeight = height;
	}

	/*
	 * 获得图片大小 传入参数 String path ：图片路径
	 */
	public long getPicSize(String path) {
		file = new File(path);
		return file.length();
	}

	// 图片处理
	public boolean compressPic() {
		try {
			// 获得源文件
			file = new File(inputDir + inputFileName);
			if (!file.exists()) {
				System.out.println("文件不存在！");
				return false;
			}
			Image img = ImageIO.read(file);
			// 判断图片格式是否正确
			if (img.getWidth(null) == -1) {
				System.out.println("图片格式不正确！");
				return false;
			} else {
				int newWidth;
				int newHeight;
				// 判断是否是等比缩放
				if (this.proportion == true) {
					// 为等比缩放计算输出的图片宽度及高度
					double rate1 = ((double) img.getWidth(null)) / (double) outputWidth + 0.1;
					double rate2 = ((double) img.getHeight(null)) / (double) outputHeight + 0.1;
					// 根据缩放比率大的进行缩放控制
					double rate = rate1 > rate2 ? rate1 : rate2;
					newWidth = (int) (((double) img.getWidth(null)) / rate);
					newHeight = (int) (((double) img.getHeight(null)) / rate);
				} else {
					newWidth = outputWidth; // 输出的图片宽度
					newHeight = outputHeight; // 输出的图片高度
				}
				BufferedImage tag = new BufferedImage((int) newWidth, (int) newHeight, BufferedImage.TYPE_INT_RGB);

				/*
				 * Image.SCALE_SMOOTH 的缩略算法 生成缩略图片的平滑度的 优先级比速度高 生成的图片质量比较好 但速度慢
				 */
				tag.getGraphics().drawImage(img.getScaledInstance(newWidth, newHeight, Image.SCALE_SMOOTH), 0, 0, null);
				FileOutputStream out = new FileOutputStream(outputDir + outputFileName);
				// JPEGImageEncoder可适用于其他图片类型的转换
				JPEGImageEncoder encoder = JPEGCodec.createJPEGEncoder(out);
				encoder.encode(tag);
				out.close();
			}
		} catch (IOException ex) {
			ex.printStackTrace();
		}
		return true;
	}

	/**
	 * 输出流相关
	 * @param bigInputStream
	 * @param width
	 * @param height
	 * @param proportion
	 * @param smallPicPath
	 * @return
	 */
	public static byte[] compressPic(byte[] bytes,int width,int height) {
		try {
			InputStream inStream = new ByteArrayInputStream(bytes);
			// 获得源文件
			Image img = ImageIO.read(inStream);
			// 判断图片格式是否正确
			if (img.getWidth(null) == -1) {
				System.out.println("图片格式不正确！");
			} else {
				if(img.getHeight(null)>width&&img.getWidth(null)>height){//加判断，只对超过次尺寸的图片进行处理
					System.out.println("图像的高度是：  "+img.getHeight(null)+" 图像的宽度是："+img.getWidth(null));
					int newWidth,newHeight;
					double rate1 = ((double) img.getWidth(null)) / (double) width + 0.1;
					double rate2 = ((double) img.getHeight(null)) / (double) height + 0.1;
					double rate = rate1 > rate2 ? rate1 : rate2;// 根据缩放比率大的进行缩放控制
					newWidth = (int) (((double) img.getWidth(null)) / rate);
					newHeight = (int) (((double) img.getHeight(null)) / rate);
					BufferedImage tag = new BufferedImage((int) newWidth, (int) newHeight, BufferedImage.TYPE_INT_RGB);
					tag.getGraphics().drawImage(img.getScaledInstance(newWidth, newHeight, Image.SCALE_SMOOTH), 0, 0, null);
					ByteArrayOutputStream out = new ByteArrayOutputStream();  
					ImageIO.write(tag, "png", out);
					return out.toByteArray();
				}
			}
		} catch (Exception ex) {
			System.out.println("JPEG图片的ICC信息可能已经被破坏了，所以ImageIO会抛出异常，猜测此图片是p过的图片");
			ex.printStackTrace();
		}
		return bytes;
	}
	
	public boolean compressPic(String inputDir, String outputDir, String inputFileName, String outputFileName) {
		// 输入图路径
		this.inputDir = inputDir;
		// 输出图路径
		this.outputDir = outputDir;
		// 输入图文件名
		this.inputFileName = inputFileName;
		// 输出图文件名
		this.outputFileName = outputFileName;
		return compressPic();
	}

	public boolean compressPic(String inputDir, String outputDir, String inputFileName, String outputFileName, int width, int height, boolean gp) {
		// 输入图路径
		this.inputDir = inputDir;
		// 输出图路径
		this.outputDir = outputDir;
		// 输入图文件名
		this.inputFileName = inputFileName;
		// 输出图文件名
		this.outputFileName = outputFileName;
		// 设置图片长宽
		setWidthAndHeight(width, height);
		// 是否是等比缩放 标记
		this.proportion = gp;
		return compressPic();
	}

	// main测试
	// compressPic(大图片路径,生成小图片路径,大图片文件名,生成小图片文名,生成小图片宽度,生成小图片高度,是否等比缩放(默认为true))
	public static void main(String[] arg) throws Exception {
		//ImageUtils mypic = new ImageUtils();
		//System.out.println("输入的图片大小：" + mypic.getPicSize("f:\\big.jpg") / 1024 + "KB");
		InputStream stream=null;
		int start = (int) System.currentTimeMillis(); // 开始时间
		File dirFile=new File("G:\\百度云同步盘\\临时下载\\1photo\\1photo\\");
		if(dirFile.isDirectory()){
			File[] allPhone=dirFile.listFiles();
			for(File phone:allPhone){
				stream=new FileInputStream(phone);
				try{
					byte[] tt=ImageUtils.compressPic(FileUtils.readBytes(stream),320,480);
				}catch(Exception e){
					System.out.println("phone name is"+phone.getName());
					e.printStackTrace();
				}
				/*FileOutputStream outputStream=new FileOutputStream("G:\\百度云同步盘\\临时下载\\smart\\"+phone.getName());
				outputStream.write(tt);
				outputStream.flush();
				outputStream.close();*/
			}
		}
		/*File file=new File("f:\\big.jpg");
		File file2=new File("f:\\small3.jpg");*/
		//byte[] tt=ImageUtils.compressPic(stream,320,480);
		int end = (int) System.currentTimeMillis(); // 结束时间
		System.out.println("总共用了：" + (end - start) + "毫秒");
	}
}
